import os
import xlsxwriter
from openpyxl import load_workbook

class CreateExcel(object):
    def __init__(self):
        super(CreateExcel, self).__init__()


    def create(self,sheet_name,data_list,path):
        """
        从tab里传来的数据信息列表，有多少个分列就得有多少个元素。
        可以出个接口，直接将qt的tab转为excel,无需考虑具体内容
        :param data_list: [asset_data,shot_data]
                         asset_data=[['难度等级','个数(个)','单价(个/元)'，合计],
                                     ['资产','A1','12','1000','12000'],
                                     ...]
                         shot_data=[]
        :param sheet_name:
        :param path
        :return:
        """
        wb=xlsxwriter.Workbook(path)
        ws=wb.add_worksheet(sheet_name)
        head_format = wb.add_format(
            {'font_size': 12, 'bold': 1, 'align': 'center', 'valign': 'vcenter', 'fg_color': '#ADD8E6', 'left': 1,
             'right': 1})
        content_format = wb.add_format(
            {'font_size': 10, 'bold': 0, 'align': 'center', 'valign': 'vcenter', 'top': 1, 'left': 1, 'right': 1,
             'bottom': 1})
        for i in range(len(data_list)):
            ws.write_row(i,0,data_list[i])
            ws.set_row(i,25)
        wb.close()